# 定义GPU数量变量
NUM_GPUS=2

#训练代码
python train_net_lmpm.py \
    --config-file configs/gvrsv1_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto \
    --tlggn_k 5 \
    MODEL.WEIGHTS ./weights/model_final_86143f.pkl \
    SOLVER.IMS_PER_BATCH $NUM_GPUS \
    OUTPUT_DIR ./outputs/gvrsv1k5
 
sleep 15

#推理代码
python train_net_lmpm.py \
    --config-file configs/gvrsv1_SWIN_bs8.yaml \
    --num-gpus $NUM_GPUS --dist-url auto --eval-only \
    --tlggn_k 5 \
    MODEL.WEIGHTS ./outputs/gvrsv1k5/model_final.pth \
    SOLVER.IMS_PER_BATCH $NUM_GPUS \
    OUTPUT_DIR ./outputs/gvrsv1k5